# 导包
import app
import requests

# 创建接口类
class EmployeeAPI:
    # 初始化
    def __init__(self):
        self.add_employee_url = app.BESE_URL + "/api/sys/user"
        self.update_employee_url = app.BESE_URL + "/api/sys/user/{}"
        self.get_employee_url = app.BESE_URL + "/api/sys/user/{}"
        self.delete_employee_url = app.BESE_URL + "/api/sys/user/{}"

    # 员工添加
    def add_employee(self, add_employee_data):
        return requests.post(url=self.add_employee_url, json=add_employee_data, headers=app.headers_data)

    # 员工修改
    def update_employee(self, employee_id, update_employee_data):
        url = self.update_employee_url.format(employee_id)
        return requests.put(url=url, json=update_employee_data, headers=app.headers_data)

    # 员工查询
    def get_employee(self, employee_id):
        url = self.get_employee_url.format(employee_id)
        return requests.get(url=url, headers=app.headers_data)

    # 员工删除
    def delete_employee(self, employee_id):
        url = self.delete_employee_url.format(employee_id)
        return requests.delete(url=url, headers=app.headers_data)

